home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_marshal.py < prev    next >
Encoding:
Python Source  |  2005-10-18  |  7.0 KB  |  199 lines
  1. #!/usr/bin/env python
  2. # -*- coding: iso-8859-1 -*-
  3.  
  4. from test import test_support
  5. import marshal
  6. import sys
  7. import unittest
  8. import os
  9.  
  10. class IntTestCase(unittest.TestCase):
  11.     def test_ints(self):
  12.         # Test the full range of Python ints.
  13.         n = sys.maxint
  14.         while n:
  15.             for expected in (-n, n):
  16.                 s = marshal.dumps(expected)
  17.                 got = marshal.loads(s)
  18.                 self.assertEqual(expected, got)
  19.                 marshal.dump(expected, file(test_support.TESTFN, "wb"))
  20.                 got = marshal.load(file(test_support.TESTFN, "rb"))
  21.                 self.assertEqual(expected, got)
  22.             n = n >> 1
  23.         os.unlink(test_support.TESTFN)
  24.  
  25.     def test_int64(self):
  26.         # Simulate int marshaling on a 64-bit box.  This is most interesting if
  27.         # we're running the test on a 32-bit box, of course.
  28.  
  29.         def to_little_endian_string(value, nbytes):
  30.             bytes = []
  31.             for i in range(nbytes):
  32.                 bytes.append(chr(value & 0xff))
  33.                 value >>= 8
  34.             return ''.join(bytes)
  35.  
  36.         maxint64 = (1L << 63) - 1
  37.         minint64 = -maxint64-1
  38.  
  39.         for base in maxint64, minint64, -maxint64, -(minint64 >> 1):
  40.             while base:
  41.                 s = 'I' + to_little_endian_string(base, 8)
  42.                 got = marshal.loads(s)
  43.                 self.assertEqual(base, got)
  44.                 if base == -1:  # a fixed-point for shifting right 1
  45.                     base = 0
  46.                 else:
  47.                     base >>= 1
  48.  
  49.     def test_bool(self):
  50.         for b in (True, False):
  51.             new = marshal.loads(marshal.dumps(b))
  52.             self.assertEqual(b, new)
  53.             self.assertEqual(type(b), type(new))
  54.             marshal.dump(b, file(test_support.TESTFN, "wb"))
  55.             new = marshal.load(file(test_support.TESTFN, "rb"))
  56.             self.assertEqual(b, new)
  57.             self.assertEqual(type(b), type(new))
  58.  
  59. class FloatTestCase(unittest.TestCase):
  60.     def test_floats(self):
  61.         # Test a few floats
  62.         small = 1e-25
  63.         n = sys.maxint * 3.7e250
  64.         while n > small:
  65.             for expected in (-n, n):
  66.                 f = float(expected)
  67.                 s = marshal.dumps(f)
  68.                 got = marshal.loads(s)
  69.                 self.assertEqual(f, got)
  70.                 marshal.dump(f, file(test_support.TESTFN, "wb"))
  71.                 got = marshal.load(file(test_support.TESTFN, "rb"))
  72.                 self.assertEqual(f, got)
  73.             n /= 123.4567
  74.  
  75.         f = 0.0
  76.         s = marshal.dumps(f)
  77.         got = marshal.loads(s)
  78.         self.assertEqual(f, got)
  79.  
  80.         n = sys.maxint * 3.7e-250
  81.         while n < small:
  82.             for expected in (-n, n):
  83.                 f = float(expected)
  84.                 s = marshal.dumps(f)
  85.                 got = marshal.loads(s)
  86.                 self.assertEqual(f, got)
  87.                 marshal.dump(f, file(test_support.TESTFN, "wb"))
  88.                 got = marshal.load(file(test_support.TESTFN, "rb"))
  89.                 self.assertEqual(f, got)
  90.             n *= 123.4567
  91.         os.unlink(test_support.TESTFN)
  92.  
  93. class StringTestCase(unittest.TestCase):
  94.     def test_unicode(self):
  95.         for s in [u"", u"AndrΦ Previn", u"abc", u" "*10000]:
  96.             new = marshal.loads(marshal.dumps(s))
  97.             self.assertEqual(s, new)
  98.             self.assertEqual(type(s), type(new))
  99.             marshal.dump(s, file(test_support.TESTFN, "wb"))
  100.             marshal.load(file(test_support.TESTFN, "rb"))
  101.             self.assertEqual(s, new)
  102.             self.assertEqual(type(s), type(new))
  103.         os.unlink(test_support.TESTFN)
  104.  
  105.     def test_string(self):
  106.         for s in ["", "AndrΦ Previn", "abc", " "*10000]:
  107.             new = marshal.loads(marshal.dumps(s))
  108.             self.assertEqual(s, new)
  109.             self.assertEqual(type(s), type(new))
  110.             marshal.dump(s, file(test_support.TESTFN, "wb"))
  111.             marshal.load(file(test_support.TESTFN, "rb"))
  112.             self.assertEqual(s, new)
  113.             self.assertEqual(type(s), type(new))
  114.         os.unlink(test_support.TESTFN)
  115.  
  116.     def test_buffer(self):
  117.         for s in ["", "AndrΦ Previn", "abc", " "*10000]:
  118.             b = buffer(s)
  119.             new = marshal.loads(marshal.dumps(b))
  120.             self.assertEqual(s, new)
  121.             marshal.dump(b, file(test_support.TESTFN, "wb"))
  122.             marshal.load(file(test_support.TESTFN, "rb"))
  123.             self.assertEqual(s, new)
  124.         os.unlink(test_support.TESTFN)
  125.  
  126. class ExceptionTestCase(unittest.TestCase):
  127.     def test_exceptions(self):
  128.         new = marshal.loads(marshal.dumps(StopIteration))
  129.         self.assertEqual(StopIteration, new)
  130.  
  131. class CodeTestCase(unittest.TestCase):
  132.     def test_code(self):
  133.         co = ExceptionTestCase.test_exceptions.func_code
  134.         new = marshal.loads(marshal.dumps(co))
  135.         self.assertEqual(co, new)
  136.  
  137. class ContainerTestCase(unittest.TestCase):
  138.     d = {'astring': 'foo@bar.baz.spam',
  139.          'afloat': 7283.43,
  140.          'anint': 2**20,
  141.          'ashortlong': 2L,
  142.          'alist': ['.zyx.41'],
  143.          'atuple': ('.zyx.41',)*10,
  144.          'aboolean': False,
  145.          'aunicode': u"AndrΦ Previn"
  146.          }
  147.     def test_dict(self):
  148.         new = marshal.loads(marshal.dumps(self.d))
  149.         self.assertEqual(self.d, new)
  150.         marshal.dump(self.d, file(test_support.TESTFN, "wb"))
  151.         marshal.load(file(test_support.TESTFN, "rb"))
  152.         self.assertEqual(self.d, new)
  153.         os.unlink(test_support.TESTFN)
  154.  
  155.     def test_list(self):
  156.         lst = self.d.items()
  157.         new = marshal.loads(marshal.dumps(lst))
  158.         self.assertEqual(lst, new)
  159.         marshal.dump(lst, file(test_support.TESTFN, "wb"))
  160.         marshal.load(file(test_support.TESTFN, "rb"))
  161.         self.assertEqual(lst, new)
  162.         os.unlink(test_support.TESTFN)
  163.  
  164.     def test_tuple(self):
  165.         t = tuple(self.d.keys())
  166.         new = marshal.loads(marshal.dumps(t))
  167.         self.assertEqual(t, new)
  168.         marshal.dump(t, file(test_support.TESTFN, "wb"))
  169.         marshal.load(file(test_support.TESTFN, "rb"))
  170.         self.assertEqual(t, new)
  171.         os.unlink(test_support.TESTFN)
  172.  
  173. class BugsTestCase(unittest.TestCase):
  174.     def test_bug_5888452(self):
  175.         # Simple-minded check for SF 588452: Debug build crashes
  176.         marshal.dumps([128] * 1000)
  177.  
  178.     def test_patch_873224(self):
  179.         self.assertRaises(Exception, marshal.loads, '0')
  180.         self.assertRaises(Exception, marshal.loads, 'f')
  181.         self.assertRaises(Exception, marshal.loads, marshal.dumps(5L)[:-1])
  182.  
  183.     def test_version_argument(self):
  184.         # Python 2.4.0 crashes for any call to marshal.dumps(x, y)
  185.         self.assertEquals(marshal.loads(marshal.dumps(5, 0)), 5)
  186.         self.assertEquals(marshal.loads(marshal.dumps(5, 1)), 5)
  187.  
  188. def test_main():
  189.     test_support.run_unittest(IntTestCase,
  190.                               FloatTestCase,
  191.                               StringTestCase,
  192.                               CodeTestCase,
  193.                               ContainerTestCase,
  194.                               ExceptionTestCase,
  195.                               BugsTestCase)
  196.  
  197. if __name__ == "__main__":
  198.     test_main()
  199.